import Generation,UMap,Clustering,Decoding,InternalBallistic,Analysis
from DDPM.Evaluation.Control import *
from DEAD.AutoDecoder.Config import model_load_path as dead_model_load_path

if __name__ == "__main__":
    '''
    control_params=get_control_params(question_name="K阶跃装药",
        postfix="_epoch50000",
        cluster_num=5,
        guid_w=0.5)

    Generation.start_generation(control_params)
    UMap.start_umap(control_params)
    Clustering.start_post_processing(control_params, dead_model_load_path)
    Decoding.start_decoding(control_params)
    InternalBallistic.start_internal_ballistic_cal(control_params)
    Analysis.start_analysis(control_params)
    
    '''
    control_params=get_control_params(question_name="无条件",
        postfix="_epoch50000",
        cluster_num=50,
        guid_w=0.0)
    
    Generation.start_generation(control_params)
    UMap.start_umap(control_params)
    Clustering.start_post_processing(control_params, dead_model_load_path,need_performance_evaluation=False)
    Decoding.start_decoding(control_params)
    InternalBallistic.start_internal_ballistic_cal(control_params)
    Analysis.start_analysis(control_params)
